import torch
from torch.utils.data import DataLoader
from torchvision import datasets
import torchvision.transforms as transforms

batch_size = 64


train_transform = transforms.Compose([transforms.Resize([224, 224]), transforms.ToTensor()])
valid_transform = transforms.Compose([transforms.Resize([224, 224]), transforms.ToTensor()])
test_transform = transforms.Compose([transforms.Resize([224, 224]), transforms.ToTensor()])

